package com.jw.debezuim.utils;

import lombok.extern.slf4j.Slf4j;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.net.Inet4Address;
import java.net.InetAddress;
import java.net.NetworkInterface;
import java.net.SocketException;
import java.util.Enumeration;

/**
 * @auther jw
 * @date 2022-02-21
 */
@Slf4j
public class InetAddressUtil {
    private static volatile InetAddress ipHost = null;
    public static InetAddress getInetAddress() {
        if (ipHost != null) {
            return ipHost;
        }
        synchronized (InetAddressUtil.class) {
            if (ipHost == null) {
                Enumeration<NetworkInterface> allNetInterfaces = null;
                try {
                    allNetInterfaces = NetworkInterface.getNetworkInterfaces();
                } catch (SocketException e) {
                    e.printStackTrace();
                }
                while (allNetInterfaces.hasMoreElements()) {
                    NetworkInterface netInterface = (NetworkInterface) allNetInterfaces.nextElement();
                    Enumeration<InetAddress> addresses = netInterface.getInetAddresses();
                    while (addresses.hasMoreElements()) {
                        InetAddress _inetAddress = addresses.nextElement();
                        if (_inetAddress != null && _inetAddress instanceof Inet4Address) {
                            ipHost = _inetAddress;
                            return ipHost;
                        }
                    }
                }
            }
            return ipHost = Inet4Address.getLoopbackAddress();
        }
    }
    public static String getRemoteIP() {
        try {
            HttpServletRequest request = ((ServletRequestAttributes) (RequestContextHolder.currentRequestAttributes())).getRequest();
            if (request != null){
                String ip = null;
                //X-Forwarded-For：Squid 服务代理
                String ipAddresses = request.getHeader("X-Forwarded-For");
                if (ipAddresses == null || ipAddresses.length() == 0 || "unknown".equalsIgnoreCase(ipAddresses)) {
                    //Proxy-Client-IP：apache 服务代理
                    ipAddresses = request.getHeader("Proxy-Client-IP");
                }
                if (ipAddresses == null || ipAddresses.length() == 0 || "unknown".equalsIgnoreCase(ipAddresses)) {
                    //WL-Proxy-Client-IP：weblogic 服务代理
                    ipAddresses = request.getHeader("WL-Proxy-Client-IP");
                }
                if (ipAddresses == null || ipAddresses.length() == 0 || "unknown".equalsIgnoreCase(ipAddresses)) {
                    //HTTP_CLIENT_IP：有些代理服务器
                    ipAddresses = request.getHeader("HTTP_CLIENT_IP");
                }
                if (ipAddresses == null || ipAddresses.length() == 0 || "unknown".equalsIgnoreCase(ipAddresses)) {
                    //X-Real-IP：nginx服务代理
                    ipAddresses = request.getHeader("X-Real-IP");
                }
                //有些网络通过多层代理，那么获取到的ip就会有多个，一般都是通过逗号（,）分割开来，并且第一个ip为客户端的真实IP
                if (ipAddresses != null && ipAddresses.length() != 0) {
                    ip = ipAddresses.split(",")[0];
                }
                //还是不能获取到，最后再通过request.getRemoteAddr();获取
                if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ipAddresses)) {
                    ip = request.getRemoteAddr();
                }
                return ip.equals("0:0:0:0:0:0:0:1") ? "127.0.0.1" : ip;
            }
        } catch (Exception e) {
            log.error("获取ip报错", e);
        }
        return "";
    }

}
